BaseMessageTemplates.java

/*
 * Project: UHC
 * Class: gg.uhc.uhc.messages.BaseMessageTemplates
 *
 * The MIT License (MIT)
 *
 * Copyright (c) 2015 Graham Howden <graham_howden1 at yahoo.co.uk>.
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 */

package gg.uhc.uhc.messages;

import com.github.mustachejava.DefaultMustacheFactory;
import com.github.mustachejava.Mustache;
import com.github.mustachejava.MustacheFactory;

import com.google.common.base.Function;
import com.google.common.collect.*;
import com.typesafe.config.Config;
import com.typesafe.config.ConfigValue;
import com.typesafe.config.ConfigValueType;

import java.io.StringReader;
import java.io.StringWriter;
import java.util.List;

public class BaseMessageTemplates implements MessageTemplates {

    protected final Config config;
    protected final MustacheFactory templating;
    protected final ListMultimap<String, Mustache> templates;

    public BaseMessageTemplates(Config config) {
        this.config = config;
        this.templating = new DefaultMustacheFactory();
        this.templates = ArrayListMultimap.create();
    }

    @Override
    public Config getConfig() {
        return config;
    }

    @Override
    public String getRaw(String key) {
        return config.getString(key);
    }

    @Override
    public List<String> getRawStrings(String path) {
        final ConfigValue value = config.getValue(path);

        if (value.valueType() == ConfigValueType.LIST) {
            return config.getStringList(path);
        }

        return Lists.newArrayList(config.getString(path));
    }

    @Override
    public Mustache getTemplate(String key) {
        // compile the tempalte from the config value and store it in our map
        if (!templates.containsKey(key)) {
            templates.put(key, templating.compile(new StringReader(getRaw(key)), key));
        }

        return Iterables.getFirst(templates.get(key), null);
    }

    @Override
    public List<Mustache> getTemplates(final String path) {
        // compile the template from the config value and store it in our map
        if (!templates.containsKey(path)) {
            final List<Mustache> temps = Lists.transform(getRawStrings(path), new Function<String, Mustache>() {

                protected int index;

                @Override
                public Mustache apply(String input) {
                    return templating.compile(new StringReader(input), path + ".[" + (index++) + "]");
                }
            });

            templates.putAll(path, temps);
        }

        return templates.get(path);
    }

    @Override
    public String evalTemplate(String key, Object... context) {
        final Mustache template = getTemplate(key);

        final StringWriter writer = new StringWriter();
        template.execute(writer, context);

        return writer.getBuffer().toString();
    }

    @Override
    public List<String> evalTemplates(String path, final Object... context) {
        final List<Mustache> temps = getTemplates(path);

        return Lists.transform(temps, new Function<Mustache, String>() {
            @Override
            public String apply(Mustache input) {
                final StringWriter writer = new StringWriter();
                input.execute(writer, context);

                return writer.getBuffer().toString();
            }
        });
    }

    @Override
    public MessageTemplates getRoot() {
        return this;
    }
}